#pragma once
#include <iostream>
#include <string>
#include <fstream>
#include "Log.hpp"
#include <vector>
#include <boost/algorithm/string.hpp>
#include "cppjieba/Jieba.hpp"
#include <mutex>
namespace ns_util
{
  class FileUtil
  {
    public:
      static bool ReadFile(const std::string& file_path,std::string* out)
      {
        std::ifstream in(file_path,std::ios::in);
        if(!in.is_open())
        {
          LOG(FATAL,"open file [%d->%s]",errno,strerror(errno));
          return false;
        }
        std::string line;
        while(std::getline(in,line))
        {
          *out+=line;
        }
        in.close();
        return true;
      }
  };
  class StringUtil
  {
    public:
      static void Split(const std::string& target,std::vector<std::string>*out,const std::string& sep)
      {
        //利用boost库
        boost::split(*out,target,boost::is_any_of(sep),boost::token_compress_on);
      }
  };
  const char* const DICT_PATH = "./dict/jieba.dict.utf8";
  const char* const HMM_PATH = "./dict/hmm_model.utf8";
  const char* const USER_DICT_PATH = "./dict/user.dict.utf8";
  const char* const IDF_PATH = "./dict/idf.utf8";
  const char* const STOP_WORD_PATH = "./dict/stop_words.utf8";
  class JiebaUtil
  {
    private:
      cppjieba::Jieba jieba;
      std::unordered_map<std::string,bool> stop_words;
    private:
      JiebaUtil()
        :jieba(DICT_PATH,HMM_PATH,USER_DICT_PATH,IDF_PATH,STOP_WORD_PATH)
      {}
      JiebaUtil(const JiebaUtil&) = delete ;
    public:
      static JiebaUtil* get_instance()
      {
        if(nullptr == instance)
        {
          mtx.lock();
          if(nullptr == instance)
          {
            instance = new JiebaUtil();
            instance->InitJiebaUtil();
          }
          mtx.unlock();
        }
        return instance;
      }
      void InitJiebaUtil()
      {
        std::ifstream in(STOP_WORD_PATH);
        if(!in.is_open())
        {
          LOG(FATAL,"load stop wordsfile error");
          return;
        }
        std::string line;
        while(std::getline(in,line))
        {
          stop_words.insert({line,true});
        }
        in.close();
      }
      void CutStringHelper(const std::string& src,std::vector<std::string>*out)
      {
        jieba.CutForSearch(src,*out);
        for(auto iter = out->begin();iter != out->end();)
        {
          auto it = stop_words.find(*iter);
          if(it!=stop_words.end())
          {
            //说明当前的string是暂停词，需要去除
            iter = out->erase(iter);
          }
          else
          {
            iter++;
          }
        }
      }
    public:
      static void CutString(const std::string&src,std::vector<std::string>*out)
      {
        ns_util::JiebaUtil::get_instance()->CutStringHelper(src,out);
      }
    private:
      static JiebaUtil* instance;
      static std::mutex mtx;
  };
  JiebaUtil* JiebaUtil::instance = nullptr;
  std::mutex JiebaUtil::mtx;
}
